import os
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image, ImageDraw, ImageFont
import json
import glob

# 确保图片目录存在
os.makedirs("./zh_data/images", exist_ok=True)

def create_product_image(product_id, product_name, category, output_path):
    """创建示例商品图片"""
    # 创建一个空白图片
    img = Image.new('RGB', (400, 400), color=(255, 255, 255))
    draw = ImageDraw.Draw(img)
    
    # 添加产品ID
    draw.text((20, 20), f"ID: {product_id}", fill=(0, 0, 0))
    
    # 添加产品名称
    draw.text((20, 60), f"名称: {product_name}", fill=(0, 0, 0))
    
    # 添加产品类别
    draw.text((20, 100), f"类别: {category}", fill=(0, 0, 0))
    
    # 添加示例图片标记
    draw.text((20, 200), "示例图片", fill=(0, 0, 0))
    
    # 保存图片
    img.save(output_path)
    print(f"已创建图片: {output_path}")

def main():
    """主函数"""
    # 获取所有产品JSON文件
    product_files = glob.glob("./zh_data/products/*.json")
    
    for product_file in product_files:
        # 读取产品数据
        with open(product_file, 'r', encoding='utf-8') as f:
            products = json.load(f)
        
        # 为每个产品创建图片
        for product in products:
            product_id = product['id']
            product_name = product['name']
            category = product['category']
            
            # 创建图片路径
            image_path = f"./zh_data/images/{product_id}.png"
            
            # 创建图片
            create_product_image(product_id, product_name, category, image_path)

if __name__ == "__main__":
    main()
    print("所有示例图片已创建完成。")
    print("请将这些图片替换为实际的商品图片，或者使用这些示例图片进行测试。") 